cd ../dpi-uncon
nohup python train.py --data_type MNISTDataset \
    --data_args "{'power_of_two': True}" \
    --results_dir ../results/dpi/uncon --input_shape 3 32 32 \
    --post_model_type Glow --post_model_args "{'num_conv_layers':[8, 16, 16], 'num_fc_layers':[16]}" \
    --reg_parameter 1 --lr_post 1e-04 --batch_size 1 \
    --num_bits 5 --num_iters_post 200000 --gpu 3 --num_z 1 \
    --degradation_type GaussianNoise \
    --degradation_args "{'mean':0., 'std':0}" \
    --reg_tv 0 \
    --num_iters 5000 > /home/baiweimin/yifei/flow-diff/log/flow/dpi_uncon_afhq_test1.log 2>&1 &

    # --degradation_type GaussianBlurNoise \
    # --degradation_args "{'kernel_sigma':1, 'mean':0., 'std':0.01}" \